finite difference method in python